package com.gzist.aidemo.entity.model;

import com.gzist.aidemo.common.ModelConstants;
import lombok.Data;

/**
 * 图片请求体封装
 */
@Data
public class ImageGenerationRequest {
    /**
     * 模型名称
     */
    private String model = ModelConstants.KWAI_KOLORS;
    /**
     * 提示词
     */
    private String prompt;
    /**
     * 负提示词
     */
    private String negative_prompt;
    /**
     * 图片尺寸
     */
    private String image_size = "1024x1024";
    /**
     * 输出图片数量
     */
    private int batch_size = 1;
    /**
     * 随机种子值
     */
    private String seed = String.valueOf(Integer.MAX_VALUE);
    /**
     * 推理步数(越大图片质量越高，时间越久)
     */
    private int num_inference_steps = 20;
    /**
     * 引导值(越高越符合提示词)
     */
    private double guidance_scale = 7.5;
    /**
     * 参考图片(Base64编码格式)
     */
    private String image;
}